package _57_二叉树的下一个结点;
/*
题目
给定一个二叉树和其中的一个结点，请找出中序遍历顺序的下一个结点并且返回。
注意，树中的结点不仅包含左右子结点，同时包含指向父结点的指针。
 */
class TreeLinkNode {
    int val;
    TreeLinkNode left = null;
    TreeLinkNode right = null;
    TreeLinkNode next = null;

    TreeLinkNode(int val) {
        this.val = val;
    }
}/*
链接：https://www.nowcoder.com/questionTerminal/9023a0c988684a53960365b889ceaf5e
来源：牛客网

1.二叉树为空，则返回空；
2.节点右孩子存在，则设置一个指针从该节点的右孩子出发，一直沿着指向左子结点的指针找到的叶子节点即为下一个节点；
3.节点不是根节点。如果该节点是其父节点的左孩子，则返回父节点；否则继续向上遍历其父节点的父节点，重复之前的判断，返回结果。代码如下：*/
public class Solution {
    public TreeLinkNode GetNext(TreeLinkNode pNode)
    {
        if(pNode==null){
            return null;
        }
        //先看有没有右结点
        if(pNode.right!=null){
            pNode = pNode.right;
            //看有没有找到最左结点
            while(pNode.left!=null){
                pNode = pNode.left;
            }
            return pNode;
        }
        else {
            // 父节点不为空
            while(pNode.next!=null){
                TreeLinkNode root = pNode.next;
                //如果是父节点的左孩子，返回父节点
                if(root.left ==pNode) return root;
                //否则变为父节点
                else{
                    pNode = pNode.next;
                }
            }
        }
        return null;
    }
}
